package com.itheima.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.OrderItem;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.itheima.mapper.ClazzMapper;
import com.itheima.mapper.EmpMapper;
import com.itheima.mapper.StudentMapper;
import com.itheima.pojo.Clazz;
import com.itheima.pojo.Emp;
import com.itheima.pojo.PageResult;
import com.itheima.pojo.Student;
import com.itheima.service.ClazzService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;

@Service
public class ClazzServiceImpl extends ServiceImpl<ClazzMapper, Clazz>  implements ClazzService {



    @Autowired
    private ClazzMapper clazzMapper;

    @Autowired
    private EmpMapper empMapper;

    @Autowired
    private StudentMapper studentMapper;


    /**
     * 班级分页查询
     *
     * @param name
     * @param begin
     * @param end
     * @param page
     * @param pageSize
     */
    @Override
    public PageResult<Clazz> getPageResult(String name, LocalDate begin, LocalDate end, int page, int pageSize) {
        // 创建分页对象
        Page<Clazz> pageParam = Page.of(page, pageSize);

        // 按修改时间倒序排序
        pageParam.addOrder(OrderItem.desc("update_time"));

        // 创建条件构造器对象
        LambdaQueryWrapper<Clazz> lambdaQuery = new LambdaQueryWrapper<>();

        // 添加查询条件
        if (name != null && !name.isEmpty()) {
            lambdaQuery.apply("BINARY name LIKE {0}", "%" + name + "%"); // 区分大小写
        }
        lambdaQuery.ge(begin != null, Clazz::getBeginDate, begin)
                .le(end != null, Clazz::getEndDate, end);

        // 调用分页查询方法，传入分页对象和条件构造器
        Page<Clazz> clazzPage = clazzMapper.selectPage(pageParam, lambdaQuery);

        // 获取分页数据中的记录
        List<Clazz> clazzList = clazzPage.getRecords();

        // 获取所有班级的班主任ID、开始时间和结束时间
        List<Integer> masterIds = clazzList.stream().map(Clazz::getMasterId).toList();
        List<LocalDate> beginTimes = clazzList.stream().map(Clazz::getBeginDate).toList();
        List<LocalDate> endTimes = clazzList.stream().map(Clazz::getEndDate).toList();

        // 批量查询这些班主任的信息
        List<Emp> emps = empMapper.selectBatchIds(masterIds);

        // 将班主任姓名封装到班级对象中，并判断班级状态
        clazzList.forEach(clazz -> {
            clazz.setStatus(clazz.getBeginDate().isAfter(LocalDate.now()) ? "未开课" :
                    clazz.getEndDate().isAfter(LocalDate.now()) ? "进行中" : "已结课");
        });
        clazzList.forEach(clazz -> {
            emps.forEach(emp -> {
                if (clazz.getMasterId() != null && clazz.getMasterId().equals(emp.getId())) {
                    clazz.setMasterName(emp.getName());
                }
            });
        });

        // 封装结果并返回
        return new PageResult<>(clazzPage.getTotal(), clazzList);
    }



    /**
     * 保存班级信息
     *
     * @param clazz
     */
    @Override
    public void saveClazz(Clazz clazz) {

        //添加修改时间
        clazz.setUpdateTime(LocalDateTime.now());

        //保存班级的基本信息
        save(clazz);

    }

    /**
     * 根据id查询班级信息
     *
     * @param id
     */
    @Override
    public Clazz getByIdClazz(Integer id) {
        //查询班级的基本信息
        Clazz clazz = getById(id);

        //封装返回
        return clazz;
    }


    /**
     * 更新班级信息
     *
     * @param clazz
     */
    @Override
    public void updateClazz(Clazz clazz) {
        //调用service层方法修改班级信息
        //根据id更新班级的基本信息
        clazz.setUpdateTime(LocalDateTime.now());
        updateById(clazz);
    }

    /**
     * 根据id删除班级信息
     *
     * @param id
     */
    @Override
    public void removeByIdClazz(Integer id) {
        // 查询该班级下是否存在学员
        LambdaQueryWrapper<Student> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Student::getClazzId, id); // 假设 Student 表中有一个 clazzId 字段表示班级ID
        Long studentCount = studentMapper.selectCount(queryWrapper);

        // 如果存在学员，抛出异常或返回提示信息
        if (studentCount > 0) {
            throw new RuntimeException("该班级下存在学员，无法删除！");
        }

        // 如果不存在学员，删除班级
        removeById(id);
    }

}
